#pragma once
#include <iostream>
#include <sys/types.h>
#include <sys/ipc.h>
#include <sys/shm.h>
#include <unistd.h>

const int shmid = -1;
const int gsize = 1024;
const std::string PATH = "."; // 定义文件路径
const int proj_id = 0x66;
const int gmode = 0666; // 定义文件权限

#define CREATER "ceater" // 创建者
#define USER "user"      // 使用者
#define ERR_EXIT(m)         \
    do                      \
    {                       \
        perror(m);          \
        exit(EXIT_FAILURE); \
    } while (0)

// 先描述 描述共享内存特性
// 集成化创建者和使用者的功能  创建者：创建 删除 映射 得到虚拟地址起始位置  使用者: 获取 映射 得到虚拟地址起始位置
class Shm
{
private:
    void CreatHelper(int flag)
    {
        // 创建共享内存
        _shmid = shmget(_key, _size, flag);
        // 如果创建失败 程序终止 并打印报错信息
        if (_shmid < 0)
        {
            ERR_EXIT("shmget");
        }
        std::cout << "shmid的值为:" << _shmid << std::endl;
    }

    // 创建共享内存
    void Creat()
    {
        CreatHelper(IPC_CREAT | IPC_EXCL | gmode);
    }

    // 获取共享内存
    void Get()
    {
        CreatHelper(IPC_CREAT | gmode);
    }

    // 做共享内存映射(挂载)
    void Map()
    {
        _start_mem = shmat(_shmid, nullptr, 0);
        if (_start_mem == (void *)-1)
        {
            ERR_EXIT("shmat");
        }
        std::cout << "挂载成功" << std::endl;
    }

    // 解除共享内存映射(卸载)
    void Unmap()
    {
        int n = shmdt(_start_mem);
        if (n < 0)
        {
            ERR_EXIT("shmdt");
        }
    }
    // 删除共享内存
    void Destory()
    {
        if (_user_type == CREATER)
        {
            if (_shmid == -1)
            {
                return;
            }

            // 解除映射
            Unmap();

            // 删除
            int n = shmctl(_shmid, IPC_RMID, nullptr);
            std::cout << "删除成功" << std::endl;
            if (n < 0)
            {
                ERR_EXIT("shmctl");
            }
        }
    }

public:
    // 构造函数 需要通过 文件路径 和 proj_id 指定同一份共享内存
    Shm(const std::string &pathname, int proj_id, const std::string user_type)
        : _shmid(shmid), _size(gsize), _start_mem(nullptr), _user_type(user_type)
    {
        // 初始化_key
        _key = ftok(pathname.c_str(), proj_id);
        if (_key < 0)
        {
            // 程序终止 并打印报错信息
            ERR_EXIT("ftok");
        }
        printf("key的值为:0x%x\n", _key);

        // 通过_user_type判断是创建者还是使用者
        if (user_type == CREATER)
        {
            Creat();
        }
        else if (user_type == USER)
        {
            Get();
        }

        else
        {
            std::cout << "用户类型错误" << std::endl;
            exit(EXIT_FAILURE);
        }

        // 共享内存映射
        Map();
    }

    // 获取共享内存起始地址
    void *VirtualAddress()
    {
        return _start_mem;
    }

    int mem_size()
    {
        return _size;
        ;
    }

    ~Shm()
    {
        // 释放共享内存
        Destory();
    }

private:
    int _shmid;             // 标识符
    int _size;              // 共享内寸大小
    key_t _key;             // 区分符
    void *_start_mem;       // 共享内存起始地址
    std::string _user_type; // 使用者类型
};